package org.example.myleet.p863;

import org.example.myleet.Utils.Assert;
import org.example.myleet.Utils.Parser;
import org.example.myleet.Utils.TreeNode;

import java.util.Arrays;

public class Main {
    public static void main(String[] args) {
        Solution solution = new Solution();
        Assert.isTrue(Assert.isContainPass(Arrays.asList(7,4,1), solution.distanceK(Parser.parse(new Integer[]{3,5,1,6,2,0,8,null,null,7,4}), new TreeNode(5), 2)));
        Assert.isTrue(Assert.isContainPass(Arrays.asList(5), solution.distanceK(Parser.parse(new Integer[]{3,5,1,6,2,0,8,null,null,7,4}), new TreeNode(1), 2)));
        Assert.isTrue(Assert.isContainPass(Arrays.asList(2,3), solution.distanceK(Parser.parse(new Integer[]{3,5,1,6,2,0,8,null,null,7,4}), new TreeNode(6), 2)));
    }
}
